"""
需求：用户输入目录或者文件名 ，根据用户输入的内容删除
"""
from pathlib import Path
import shutil


path = Path(input('请输入要删除的文件'))

# 如果文件不存在，提示并结束程序
if not path.exists():
    print(f'{path}路径不存在')
else:
    # 判断 是文件还是目录，
    if path.is_file():
        path.unlink()
        print(f'{path}被删除')
    elif path.is_dir():
        shutil.rmtree(path)
        print(f'{path}被删除')


